package com.utility.aes;

import java.security.SecureRandom;

import javax.crypto.Cipher;
import javax.crypto.KeyGenerator;
import javax.crypto.SecretKey;
import javax.crypto.spec.IvParameterSpec;
import javax.crypto.spec.SecretKeySpec;

import org.apache.commons.codec.binary.Base64;

public class AESUtility {
	/**
	 * 加密方法名称
	 */
	private static final String algorithm = "AES";
	/**
	 * 默认编码方式
	 */
	private static final String charset = "UTF-8";
	/**
     * 加密/解密算法-工作模式-填充模式
     */
	private static final String CIPHER_ALGORITHM = "AES/CBC/PKCS5Padding";
    /**
	 * 使用CBC模式，需要一个向量iv，可增加加密算法的强度
	 */
	private static final String IVSTR = "0201080306050704";
	/**
	 * AES加密方法
	 * @param key 加密密钥（需要和解密密钥一致）
	 * @param source 待加密的字符串
	 * @return
	 */
	public static String encrypt(String key, String source) {
		try {
            //1.构造密钥生成器，指定为AES算法,不区分大小写
            KeyGenerator keyGenerator = KeyGenerator.getInstance(algorithm);
            IvParameterSpec ivParameter = new IvParameterSpec(IVSTR.getBytes());
            //2.根据ecnodeRules规则初始化密钥生成器
            //生成一个128位的随机源,根据传入的字节数组
            keyGenerator.init(128, new SecureRandom(key.getBytes()));
            //3.产生原始对称密钥
            SecretKey originalKey = keyGenerator.generateKey();
            //4.获得原始对称密钥的字节数组
            byte [] originalKeyByte = originalKey.getEncoded();
            //5.根据字节数组生成AES密钥
            SecretKey secretKey = new SecretKeySpec(originalKeyByte, algorithm);
            //6.根据指定算法AES自成密码器
            Cipher cipher = Cipher.getInstance(CIPHER_ALGORITHM);
            //7.初始化密码器，第一个参数为加密(Encrypt_mode)或者解密解密(Decrypt_mode)操作，第二个参数为使用的KEY
            cipher.init(Cipher.ENCRYPT_MODE, secretKey,ivParameter);
            //8.获取加密内容的字节数组(这里要设置为utf-8)不然内容中如果有中文和英文混合中文就会解密为乱码
            byte [] sourceByte = source.getBytes(charset);
            //9.根据密码器的初始化方式--加密：将数据加密
            byte [] encryptByte = cipher.doFinal(sourceByte);
            //10.将加密后的数据转换为字符串
            return new String(new Base64().encode(encryptByte));
        } catch (Exception e) {
        	throw new RuntimeException(e);
        } 
	}
	
	/**
	 * AES解密方法
	 * @param key 解密密钥（需要和加密密钥一致）
	 * @param source 待解密的字符串
	 * @return
	 */
	public static String decrypt(String key, String source) {
		try {
            //1.构造密钥生成器，指定为AES算法,不区分大小写
            KeyGenerator keyGenerator = KeyGenerator.getInstance(algorithm);
            IvParameterSpec ivParameter = new IvParameterSpec(IVSTR.getBytes());
            //2.根据ecnodeRules规则初始化密钥生成器
            //生成一个128位的随机源,根据传入的字节数组
            keyGenerator.init(128, new SecureRandom(key.getBytes()));
            //3.产生原始对称密钥
            SecretKey originalKey = keyGenerator.generateKey();
            //4.获得原始对称密钥的字节数组
            byte [] originalKeyByte = originalKey.getEncoded();
            //5.根据字节数组生成AES密钥
            SecretKey secretKey = new SecretKeySpec(originalKeyByte, algorithm);
            //6.根据指定算法AES自成密码器
            Cipher cipher = Cipher.getInstance(CIPHER_ALGORITHM);
            //7.初始化密码器，第一个参数为加密(Encrypt_mode)或者解密(Decrypt_mode)操作，第二个参数为使用的KEY
            cipher.init(Cipher.DECRYPT_MODE, secretKey,ivParameter);
            //8.将加密并编码后的内容解码成字节数组
            byte [] sourceByte = new Base64().decode(source);
            byte [] decryptByte = cipher.doFinal(sourceByte);
            return new String(decryptByte,charset);
        } catch (Exception e) {
        	throw new RuntimeException(e);
        } 
	}
	
	public static void main(String[] args) {
		String key = "abc";
		String s = "sadfasdf的范德萨范德萨范德萨发大水发的撒";
		String ss = encrypt(key, s);
		System.out.println(ss);
		System.out.println(decrypt(key, ss));
	}
}
